# settings/router.py
# database router to multiple database by app label
"""
这是一个数据库路由类，用于根据应用标签将请求路由到不同的数据库。

Attributes:
route_app_labels (set): 一个集合，包含需要特殊处理的应用标签名称。

Methods:
db_for_read(model, **hints): 根据模型的元数据中的app_label决定从哪个数据库读取数据。如果app_label在route_app_labels中，则从'running'数据库读取，否则从'default'数据库读取。
db_for_write(model, **hints): 根据模型的元数据中的app_label决定向哪个数据库写入数据。如果app_label在route_app_labels中，则向'running'数据库写入，否则向'default'数据库写入。
allow_relation(obj1, obj2, **hints): 判断两个对象之间是否允许建立关系，此方法始终返回None。
allow_migrate(db, app_label, model_name=None, **hints): 判断是否允许迁移指定应用的数据。如果app_label在route_app_labels中，则不允许迁移，否则允许迁移。
"""
class DatabaseRouter:
    route_app_labels = {'running'}

    def db_for_read(self, model, **hints):
        if model._meta.app_label in self.route_app_labels:
            return 'running'
        return 'default'

    def db_for_write(self, model, **hints):
        if model._meta.app_label in self.route_app_labels:
            return 'running'
        return 'default'

    def allow_relation(self, obj1, obj2, **hints):
        return None

    def allow_migrate(self, db, app_label, model_name=None, **hints):
        """
        遗留数据库中的表不允许迁移
        """
        if app_label in self.route_app_labels:
            return False
        return True